simple program using python